import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * 90. 子集 II
 * https://leetcode-cn.com/problems/subsets-ii/
 */
public class Solutions_90 {
    public static void main(String[] args) {
        int[] nums1 = {1, 2, 2};  // output: {{}, {1}, {1, 2}, {1, 2, 2}, {2}, {2, 2}}
        int[] nums2 = {0};  // output: {{}, {0}}
        int[] nums3 = {4, 4, 4, 1, 4};  // output: {{}, {4}, {4, 4}, {4, 4, 4}, {4, 4, 4, 1}, {4, 4, 4, 1, 4}, {4, 4, 4, 4}, {4, 4, 1}, {4, 4, 1, 4}, {4, 4, 4}, {4, 1}, {4, 1, 4}, {4, 4}, {1}, {1, 4}, {4}}


        List<int[]> inputs = new ArrayList<>();
        inputs.add(nums1);
        inputs.add(nums2);
        inputs.add(nums3);

        for (int[] input : inputs) {
            List<List<Integer>> result = subsetsWithDup(input);
            System.out.println(result);
        }
    }

    /**
     * 解题思路：回溯法
     * 如何排除重复？
     *  1.对数组进行升序或者降序排列
     *  2.递归时，若当前遍历元素与前一元素相同，且递归的开头元素是当前元素时（i > idx），则跳过该元素
     */
    public static List<List<Integer>> subsetsWithDup(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        List<Integer> list = new ArrayList<>();
        Arrays.sort(nums);
        backtrack(res, list, nums, 0);
        return res;
    }

    public static void backtrack(List<List<Integer>> res, List<Integer> list,
                                 int[] nums, int idx) {
        res.add(new ArrayList(list));
        for (int i = idx; i < nums.length; i++) {
            // 排除重复：与前一元素相同，且递归的开头元素是当前元素
            /*if (i > 0 && nums[i] == nums[i - 1] && !visited[i - 1]) {
                continue;
            }*/
            if (i > idx && nums[i] == nums[i - 1]) {
                continue;
            }
            // 添加元素
            list.add(nums[i]);
            backtrack(res, list, nums, i + 1);
            // 回溯操作：移除元素
            list.remove(list.size() - 1);
        }
    }
}
